### GSM8K 
## Python SDP
# Codellama

export CUDA_VISIBLE_DEVICES=0
export NCCL_P2P_DISABLE=1
export NCCL_IB_DISABLE=1


exp_name="gsm8k_python_sdp_codellama" \
train_file='data/gsm8k_python_sdp.json' \
test_file='data/gsm8k_test_set.json' \
engine='python' \
model_name_or_path='models/models--Qwen--Qwen2.5-0.5B/' \
tokenizer_name_or_path='models/models--Qwen--Qwen2.5-0.5B/' \
n_epochs='40' \
    bash reft_exps/paper_exps/SFT/_template.sh

# # Galactica
# exp_name="gsm8k_python_sdp_galactica" \
# train_file='data/gsm8k_python_sdp.json' \
# test_file='data/gsm8k_test_set.json' \
# engine='python' \
# model_name_or_path='hf_models/galactica-6.7b/' \
# tokenizer_name_or_path='hf_models/galactica-6.7b/' \
# n_epochs='40' \
#     bash exps/paper_exps/SFT/_template.sh

# ## NL
# # Codellama
# exp_name="gsm8k_nl_codellama" \
# train_file='data/gsm8k_nl.json' \
# test_file='data/gsm8k_test_set.json' \
# engine='nl' \
# model_name_or_path='hf_models/CodeLlama-7b-hf' \
# tokenizer_name_or_path='hf_models/CodeLlama-7b-hf/' \
# n_epochs='40' \
#     bash exps/paper_exps/SFT/_template.sh

# # Galactica
# exp_name="gsm8k_nl_galactica" \
# train_file='data/gsm8k_nl.json' \
# test_file='data/gsm8k_test_set.json' \
# engine='nl' \
# model_name_or_path='hf_models/galactica-6.7b/' \
# tokenizer_name_or_path='hf_models/galactica-6.7b/' \
# n_epochs='40' \
#     bash exps/paper_exps/SFT/_template.sh
